import yaml
from config.all_paths import performance_config
from config.global_variable import ENV_DICT
from utils.file_handling.ini_handling import LoadIni


def load_config():
    # 从yaml配置文件中读取load参数
    with open(performance_config,'r', encoding='utf-8') as f:
        config = yaml.safe_load(f)
    return {
        'profiles': config.get('performance_profiles', {})
    }

def get_env(env='test', role='doctor'):
    # 从ini配置文件中读取环境参数
    if env == 'test_wt':
        ENV_DICT.update(LoadIni("test_wutong.ini").getSection(role))
        ENV_DICT.update(LoadIni("test_wutong.ini").getSection('url'))
    elif env == 'pro_wt':
        ENV_DICT.update(LoadIni("pro_wutong.ini").getSection(role))
        ENV_DICT.update(LoadIni("pro_wutong.ini").getSection('url'))
    elif env == 'test_supda':
        ENV_DICT.update(LoadIni("test_supda.ini").getSection(role))
        ENV_DICT.update(LoadIni("test_supda.ini").getSection('url'))
    elif env == 'pro_supda':
        ENV_DICT.update(LoadIni("pro_supda.ini").getSection(role))
        ENV_DICT.update(LoadIni("pro_supda.ini").getSection('url'))
    elif env == 'test_weight':
        ENV_DICT.update(LoadIni("test_weight.ini").getSection(role))
        ENV_DICT.update(LoadIni("test_weight.ini").getSection('url'))
    elif env == 'pro_weight':
        ENV_DICT.update(LoadIni("pro_weight.ini").getSection(role))
        ENV_DICT.update(LoadIni("pro_weight.ini").getSection('url'))
    elif env == 'test_internet_hos':
        ENV_DICT.update(LoadIni("test_internet_hos.ini").getSection(role))
        ENV_DICT.update(LoadIni("test_internet_hos.ini").getSection('url'))
    elif env == 'pro_internet_hos':
        ENV_DICT.update(LoadIni("pro_internet_hos.ini").getSection(role))
        ENV_DICT.update(LoadIni("pro_internet_hos.ini").getSection('url'))
    elif env == 'test_TH':
        ENV_DICT.update(LoadIni("test_TH.ini").getSection(role))
        ENV_DICT.update(LoadIni("test_TH.ini").getSection('url'))
    else:
        print("请输入正确的环境参数")


if __name__ == '__main__':
    print(load_config())